import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

public class T1_TwoSum {
    /*
     * 两数之和
     * 给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出
     * 和为目标值 target 的那 两个 整数，并返回它们的数组下标。
     * 你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。
     * 你可以按任意顺序返回答案。
     * */

    public static void main(String[] args){
    }

    // 我的想法：hash，遍历一次数组，用target减去每个值，看剩下的值在不在数组里就好了
    // 说实话，这么写有点蠢
    public int[] mySolution(int[] nums, int target){
        Set<Integer> set = new HashSet<>();
        int[] res = new int[2];
        for (int i = 0; i < nums.length; i++) {
            set.add(nums[i]);
        }
        for (int i = 0; i < nums.length; i++) {
            int other = target - nums[i];
            if (set.contains(other)){
                int index = -1;
                for (int j = i+1; j < nums.length; j++) {
                    if(nums[j] == other){
                        index = j;
                    }
                }
                if(index >= 0){
                    res[0] = i;
                    res[1] = index;
                    break;
                }
            }
        }
        return res;
    }

    // 最开始想到用map来存索引，但考虑到有相等的元素，索引就不好存，就没用
    // 会这么想的原因是：我准备一开始就把所有的元素全加到map中
    // 但实际上，不用一开始全加到map里，两个数具有互补性，每遍历一个就加一个进去就好了
    public int[] solution(int[] nums, int target){
        int[] res = new int[2];
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            int other = target - nums[i];
            if(map.containsKey(other)){
                res[0] = i;
                res[1] = map.get(other);
                break;
            }
            map.put(nums[i],i);
        }
        return res;
    }
}
